/*
 * MIT License
 *
 * Copyright (c) 2024-2048 冰羽
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package cn.star.framework.excel.enhance;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.util.ArrayUtil;
import cn.star.framework.excel.convert.CellValueConverter;
import cn.star.framework.excel.converters.CharacterStringConverter;
import cn.star.framework.excel.converters.DateStringConverter;
import cn.star.framework.excel.converters.EnumEntryConverter;
import cn.star.framework.excel.converters.StringJdk8DateConverter;
import cn.star.framework.excel.converters.StringNull8BasicTypeConverter;
import cn.star.framework.excel.core.Type;
import com.alibaba.excel.converters.Converter;
import com.alibaba.excel.converters.ConverterKeyBuild;
import com.alibaba.excel.converters.ConverterKeyBuild.ConverterKey;
import com.alibaba.excel.converters.DefaultConverterLoader;
import com.alibaba.excel.enums.CellDataTypeEnum;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import lombok.extern.slf4j.Slf4j;

/**
 * 注册转换函数<br>
 *
 * @author zhaoweiping
 *     <p style='color: red'>Created on 2024-09-23 11:06:00
 * @since 3.0.0
 */
@Slf4j
public class GlobalConverterRegister {
  private static final Map<Class<?>, CellValueConverter> CELL_VALUE_CONVERTER_MAP =
      new ConcurrentHashMap<>();

  static {
    // write
    register(CharacterStringConverter.INSTANCE);
    register(DateStringConverter.INSTANCE);

    // read
    register(EnumEntryConverter.INSTANCE);

    // register(StringDateConverter.INSTANCE);
    // register(StringJdk8InstantConverter.INSTANCE);
    // register(StringJdk8LocalDateConverter.INSTANCE);
    // register(StringJdk8LocalDateTimeConverter.INSTANCE);

    register(StringJdk8DateConverter.INSTANCE);

    register(StringNull8BasicTypeConverter.INSTANCE);
  }

  @SuppressWarnings("rawtypes")
  public static void register(CellValueConverter... converters) {
    if (ArrayUtil.isNotEmpty(converters)) {
      for (CellValueConverter converter : converters) {
        CELL_VALUE_CONVERTER_MAP.put(converter.getClass(), converter);
      }
    }
  }

  @SuppressWarnings("rawtypes")
  public static List<CellValueConverter> getCellValueConverters() {
    return new ArrayList<>(CELL_VALUE_CONVERTER_MAP.values());
  }

  private static CellDataTypeEnum getCellDataType(Converter<?> converter) {
    try {
      return converter.supportExcelTypeKey();
    } catch (Exception e) {
      return null;
    }
  }

  private static void register(Converter<?> converter) {
    CellDataTypeEnum type = getCellDataType(converter);
    ConverterKey key = ConverterKeyBuild.buildKey(converter.supportJavaTypeKey(), type);
    DefaultConverterLoader.loadAllConverter().put(key, converter);
  }

  public static void register(Type type, Converter<?>... converters) {
    if (ArrayUtil.isNotEmpty(converters)) {
      register(type, CollectionUtil.toList(converters));
    }
  }

  public static void register(Type type, List<Converter<?>> converters) {
    Map<ConverterKey, Converter<?>> converterMap =
        type == Type.READ
            ? DefaultConverterLoader.loadDefaultReadConverter()
            : DefaultConverterLoader.loadDefaultWriteConverter();
    if (CollectionUtil.isNotEmpty(converters)) {
      try {
        converters.forEach(
            converter ->
                converterMap.put(
                    ConverterKeyBuild.buildKey(converter.supportJavaTypeKey()), converter));
      } catch (Exception e) {
        log.trace(
            "ignore converter register exception, type: {}, error: {}",
            e.getClass().getSimpleName(),
            e.getMessage());
      }
    }
  }

  /**
   * 获取转换函数
   *
   * @param clazz
   * @param type
   * @return
   */
  public static Converter<?> get(Class<?> clazz, CellDataTypeEnum type) {
    ConverterKey key = ConverterKeyBuild.buildKey(clazz, type);
    return DefaultConverterLoader.loadAllConverter().get(key);
  }
}
